/*
 *   Licensed to the Apache Software Foundation (ASF) under one or more
 *   contributor license agreements.  See the NOTICE file distributed with
 *   this work for additional information regarding copyright ownership.
 *   The ASF licenses this file to You under the Apache License, Version 2.0
 *   (the "License"); you may not use this file except in compliance with
 *   the License.  You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 */
package org.apache.dubbo.spring.boot.provider;

import java.util.ArrayList;
import java.util.List;

import org.apache.dubbo.config.spring.context.annotation.DubboComponentScan;
import org.apache.dubbo.config.spring.context.annotation.EnableDubbo;
import org.example2.HelloService2;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;

import com.alibaba.csp.sentinel.slots.block.RuleConstant;
import com.alibaba.csp.sentinel.slots.block.flow.FlowRule;
import com.alibaba.csp.sentinel.slots.block.flow.FlowRuleManager;

@SpringBootApplication
@EnableDubbo
@DubboComponentScan(basePackageClasses = { HelloService2.class })
public class ProviderApplication {

	public static void main(String[] args) {

		// initFlowRules();
		// start embedded zookeeper server
		// new EmbeddedZooKeeper(2181, false).start();
		SpringApplication.run(ProviderApplication.class, args);
	}

	// 初始化规则
	private static void initFlowRules() {
		List<FlowRule> rules = new ArrayList<>(); // 限流规则的集合
		FlowRule flowRule = new FlowRule();
		flowRule.setResource("org.example.HelloService:sayHello(java.lang.String)");// 资源(方法名称、接口）
		flowRule.setCount(10);// 限流阈值 qps=10
		flowRule.setGrade(RuleConstant.FLOW_GRADE_QPS);// 限流阈值类型（QPS 或并发线程数）
		// 流量控制手段（直接拒绝、Warm Up、匀速排队）
		flowRule.setControlBehavior(RuleConstant.CONTROL_BEHAVIOR_DEFAULT);
		flowRule.setLimitApp("sentinel-web");// 流控针对的调用来源，若为 default 则不区分调用来源
		rules.add(flowRule);
		FlowRuleManager.loadRules(rules);
	}
}
